package woa.tree.binary;

import java.util.ArrayList;
import java.util.List;

/**
 * 不同的二叉搜索树 II
 * 给你一个整数 n ，请你生成并返回所有由 n 个节点组成且节点值从 1 到 n 互不相同的不同 二叉搜索树 。可以按 任意顺序 返回答案。
 *
 * 输入：n = 3
 * 输出：[[1,null,2,null,3],[1,null,3,2],[2,1,3],[3,1,null,null,2],[3,2,null,1]]
 *
 * 输入：n = 1
 * 输出：[[1]]
 *
 * @author wangpeng
 * @date 2021/12/15
 */
public class GenerateTrees {

    public static List<TreeNode> generateTrees(int n) {
        if (n == 0) {
            return new ArrayList<>();
        }
        return helper(1, n);
    }

    public static List<TreeNode> helper(int start, int end) {
        List<TreeNode> rs = new ArrayList<>();
        if (start > end) {
            rs.add(null);
            return rs;
        }
        for (int i = start; i <= end; i++) {
            List<TreeNode> left = helper(start, i - 1);
            List<TreeNode> right = helper(i + 1, end);

            for (TreeNode l : left) {
                for (TreeNode r : right) {
                    TreeNode cur = new TreeNode(i);
                    cur.left = l;
                    cur.right = r;
                    rs.add(cur);
                }
            }
        }

        return rs;
    }

    public static void main(String[] args) {
        System.out.println(generateTrees(2));
    }
}
